package Logic

import (
	"fmt"
	"yz-go/Common"

	"github.com/spf13/viper"
	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/errors"
	"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
	ocr "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/ocr/v20181119"
)

func PTDistinguish(path string, imageBase *string) (rd Common.ResData) {

	// 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
	// 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
	// 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
	credential := common.NewCredential(
		viper.GetString("teng-xun.secret_id"),
		viper.GetString("teng-xun.secret_key"),
	)

	fmt.Printf("id: %s, key:%s \n", viper.GetString("teng-xun.secret_id"), viper.GetString("teng-xun.secret_key"))
	// 实例化一个client选项，可选的，没有特殊需求可以跳过
	cpf := profile.NewClientProfile()
	cpf.HttpProfile.Endpoint = "ocr.tencentcloudapi.com"
	// 实例化要请求产品的client对象,clientProfile是可选的
	client, _ := ocr.NewClient(credential, "ap-shanghai", cpf)

	// 实例化一个请求对象,每个接口都会对应一个request对象
	request := ocr.NewGeneralBasicOCRRequest()

	if imageBase != nil && *imageBase != "" { // 判断是不是一个空指针 和 有没有具体的值
		request.ImageBase64 = imageBase // 和url 二者选其一，若俩个都有 默认url
	} else {
		request.ImageUrl = common.StringPtr(path)
	}

	request.Scene = common.StringPtr("123")
	request.LanguageType = common.StringPtr("zh")
	request.IsPdf = common.BoolPtr(false)
	request.PdfPageNumber = common.Uint64Ptr(0)
	request.IsWords = common.BoolPtr(false)

	// 返回的resp是一个GeneralBasicOCRResponse的实例，与请求对象对应
	response, err := client.GeneralBasicOCR(request)
	if _, ok := err.(*errors.TencentCloudSDKError); ok {
		fmt.Printf("An API error has returned: %s", err)
		return
	}
	if err != nil {
		panic(err)
	}

	// 输出json格式的字符串回包
	fmt.Printf("%s", response.ToJsonString())

	return Common.ResData{Code: Common.SUCCESS, Msg: "", Data: response}
}
